import warnings
warnings.filterwarnings('ignore')
from tool import get_azure_endpoint,get_api_version,get_api_key
from langchain.prompts import ChatPromptTemplate
from langchain.chains import LLMChain
from langchain_openai import AzureChatOpenAI
if __name__ == '__main__':
    # 这里我们将参数temperature设置为0.0，从而减少生成答案的随机性。 # 如果你想要每次得到不一样的有新意的答案，可以尝试调整该参数。
    llm = AzureChatOpenAI(
        azure_endpoint=get_azure_endpoint().rstrip('/'),  # 移除尾部斜杠，只保留基础URL
        azure_deployment="gpt-4o",  # 重命名为 azure_deployment
        model_name="gpt-4o",
        openai_api_version=get_api_version(),  # 参数名不变
        openai_api_key=get_api_key(),
        openai_api_type="azure",
        temperature=0.0,
    )
    print(llm)

    prompt = ChatPromptTemplate.from_template("描述制造{product}的一个公司的最佳名称是什么?直接输出最佳答案")

    chain = LLMChain(llm=llm, prompt=prompt)

    product = "大号床单套装"
    response = chain.run(product)
    print(response)